import torch
# 在初始化时检查CUDA可用性
if torch.cuda.is_available():
    device_count = torch.cuda.device_count()
    print(f"Available GPUs: {device_count}")

